package com.sihan.swvp.filter;

import java.util.Enumeration;
import java.util.Map;
import java.util.Vector;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;

import org.apache.commons.lang3.StringUtils;

public class ModifyParametersWrapper extends HttpServletRequestWrapper{
	 private Map<String, String[]> parameterMap; // 所有参数的Map集合
	 
     public ModifyParametersWrapper(HttpServletRequest request) {
         super(request);
         parameterMap = request.getParameterMap();
     }

     // 重写几个HttpServletRequestWrapper中的方法
     /**
      * 获取所有参数名
      * 
      * @return 返回所有参数名
      */
     @Override
     public Enumeration<String> getParameterNames() {
         Vector<String> vector = new Vector<String>(parameterMap.keySet());
         return vector.elements();
     }

     /**
      * 获取指定参数名的值，如果有重复的参数名，则返回第一个的值 接收一般变量 ，如text类型
      * 
      * @param name
      *            指定参数名
      * @return 指定参数名的值
      */
     @Override
     public String getParameter(String name) {
         String[] results = parameterMap.get(name);
         if (results == null || results.length <= 0)
             return null;
         else {
             return xssEncode(results[0]);
         }
     }

     /**
      * 获取指定参数名的所有值的数组，如：checkbox的所有数据 
      * 接收数组变量 ，如checkobx类型
      */
     @Override
     public String[] getParameterValues(String name) {
         String[] results = parameterMap.get(name);
         if (results == null || results.length <= 0)
             return null;
         else {
             int length = results.length;
             for (int i = 0; i < length; i++) {
                 results[i] = xssEncode(results[i]);
             }
             return results;
         }
      }

    
     
     
     /**
 	 * 将容易引起xss漏洞的半角字符直接替换成全角字符
 	 * @param s
 	 * @return
 	 */
 	private String xssEncode(String s) {
 		if (s == null || "".equals(StringUtils.trim(s))) {
 			return s;
 		}
 		StringBuilder sb = new StringBuilder(s.length() + 16);
 		for (int i = 0; i < s.length(); i++) {
 			char c = s.charAt(i);
 			switch (c) {
 			case '>':
 				sb.append('＞');// 全角大于号
 				break;
 			case '<':
 				sb.append('＜');// 全角小于号
 				break;
 			case '\'':
 				sb.append('‘');// 全角单引号
 				break;
 			case '\"':
 				sb.append('“');// 全角双引号
 				break;
 			case '&':
 				sb.append('＆');// 全角
 				break;
 			case '\\':
 				sb.append('＼');// 全角斜线
 				break;
 			case '#':
 				sb.append('＃');// 全角井号
 				break;
 			default:
 				sb.append(c);
 				break;
 			}
 		}
 		System.out.println("******************"+sb.toString()+"*********************");
 		return sb.toString();
 	}

}
